package com.ucd.web.core.config;

import com.github.xiaoymin.knife4j.spring.annotations.EnableKnife4j;
import com.ucd.common.config.RuoYiConfig;
import com.ucd.common.constant.HttpStatus;
import io.swagger.annotations.ApiOperation;
import io.swagger.models.auth.In;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Import;
import org.springframework.http.HttpMethod;
import springfox.bean.validators.configuration.BeanValidatorPluginsConfiguration;
import springfox.documentation.builders.ApiInfoBuilder;
import springfox.documentation.builders.PathSelectors;
import springfox.documentation.builders.RequestHandlerSelectors;
import springfox.documentation.builders.ResponseBuilder;
import springfox.documentation.service.*;
import springfox.documentation.spi.DocumentationType;
import springfox.documentation.spi.service.contexts.SecurityContext;
import springfox.documentation.spring.web.plugins.Docket;
import springfox.documentation.swagger2.annotations.EnableSwagger2;

import java.util.ArrayList;
import java.util.List;

/**
 * Swagger2的接口配置
 *
 * @author ruoyi
 */

@EnableKnife4j
@Configuration
@EnableSwagger2
@Import(BeanValidatorPluginsConfiguration.class)
public class SwaggerConfig {
	/**
	 * 系统基础配置
	 */
	@Autowired
	private RuoYiConfig ruoyiConfig;

	//添加全局响应状态码
	List<Response> responseStatus = new ArrayList<Response>() {{
		add(new ResponseBuilder().code(String.valueOf(HttpStatus.SUCCESS)).description("操作成功").build());
		add(new ResponseBuilder().code(String.valueOf(HttpStatus.UNAUTHORIZED)).description("未授权").build());
		add(new ResponseBuilder().code(String.valueOf(HttpStatus.FORBIDDEN)).description("访问受限，授权过期").build());
		add(new ResponseBuilder().code(String.valueOf(HttpStatus.NOT_FOUND)).description("资源未找到").build());
		add(new ResponseBuilder().code(String.valueOf(HttpStatus.ERROR)).description("操作失败").build());
	}};


	/**
	 * 创建API
	 */
	@Bean
	public Docket createRestApi() {
		return new Docket(DocumentationType.OAS_30)
				.enable(true)
				// 添加全局响应状态码
				.useDefaultResponseMessages(true)
				.globalResponses(HttpMethod.GET, responseStatus)
				.globalResponses(HttpMethod.PUT, responseStatus)
				.globalResponses(HttpMethod.POST, responseStatus)
				.globalResponses(HttpMethod.DELETE, responseStatus)
				// 用来创建该API的基本信息，展示在文档的页面中（自定义展示的信息）
				.apiInfo(apiInfo())
				// 设置哪些接口暴露给Swagger展示
				.select()
				// 扫描所有有注解的api，用这种方式更灵活
				.apis(RequestHandlerSelectors.withMethodAnnotation(ApiOperation.class))
				.paths(PathSelectors.any()).build()
				/* 设置安全模式，swagger可以设置访问token */
				.securitySchemes(securitySchemes()).securityContexts(securityContexts())
				.pathMapping(ruoyiConfig.getApiPath());
	}

	/**
	 * 安全模式，这里指定token通过Authorization头请求头传递
	 */
	private List<SecurityScheme> securitySchemes() {
		List<SecurityScheme> apiKeyList = new ArrayList<SecurityScheme>();
		apiKeyList.add(new ApiKey("Authorization", "Authorization", In.HEADER.toValue()));
		return apiKeyList;
	}

	/**
	 * 安全上下文
	 */
	private List<SecurityContext> securityContexts() {
		List<SecurityContext> securityContexts = new ArrayList<>();
		securityContexts.add(SecurityContext.builder().securityReferences(defaultAuth()).operationSelector(o -> o.requestMappingPattern().matches("/.*")).build());
		return securityContexts;
	}

	/**
	 * 默认的安全上引用
	 */
	private List<SecurityReference> defaultAuth() {
		AuthorizationScope authorizationScope = new AuthorizationScope("global", "accessEverything");
		AuthorizationScope[] authorizationScopes = new AuthorizationScope[1];
		authorizationScopes[0] = authorizationScope;
		List<SecurityReference> securityReferences = new ArrayList<>();
		securityReferences.add(new SecurityReference("Authorization", authorizationScopes));
		return securityReferences;
	}

	/**
	 * 添加摘要信息
	 */
	private ApiInfo apiInfo() {
		// 用ApiInfoBuilder进行定制
		return new ApiInfoBuilder()
				// 设置标题
				.title("XXX系统_接口文档")
				// 描述
				.description("描述：bjucd api")
				// 作者信息
				.contact(new Contact(ruoyiConfig.getName(), null, null))
				// 版本
				.version("版本号:" + ruoyiConfig.getVersion()).build();
	}
}
